from itsdangerous import TimedJSONWebSignatureSerializer as Serializer

from meiduo_market import settings


def generate_verify_token(user_id):
    serializer = Serializer(settings.SECRET_KEY, 3600 * 24 * 2)
    token = serializer.dumps({'user_id': user_id})
    token = token.decode()
    return token


def check_verify_token(token):
    serializer = Serializer(settings.SECRET_KEY, 3600 * 24 * 2)
    try:
        user_id_dict = serializer.loads(token)
        user_id = user_id_dict.get('user_id')
        return user_id
    except Exception as e:
        print(e)
        return None


if __name__ == '__main__':
    token = generate_verify_token(1)
    print(token)
    user_id = check_verify_token(token)
    print(user_id)
